home *** CD-ROM | disk | FTP | other *** search
/ Komputer for Alle 2001 #11 / CD 11 (Black) - 2001.iso / FAVORG / FAVO_SRC.ZIP / FavOrgView.cpp < prev    next >
Encoding:
C/C++ Source or Header  |  2000-10-25  |  21.2 KB  |  809 lines
  1. // FAVORG Version 1.1
  2. // Copyright (c) 2000 Ziff Davis Media, Inc.
  3. // All rights reserved.
  4. // First Published in PC Magazine, US Edition, November 7, 2000.
  5. // Programmer: Patrick Philippot
  6.  
  7. #include "stdafx.h"
  8.  
  9. #include <shlobj.h>
  10. #include <afxpriv.h>
  11. #include <wininet.h>
  12.  
  13. #include "FavOrg.h"
  14. #include "OptionsDlg.h"
  15. #include "MultiselTreeCtrl.h"
  16. #include "FavOrgTreeCtl.h"
  17. #include "SelIconDlg.h"
  18.  
  19. #include "FavOrgDoc.h"
  20. #include "FavOrgView.h"
  21.  
  22. #include "mainfrm.h"
  23.  
  24. #ifdef _DEBUG
  25. #define new DEBUG_NEW
  26. #undef THIS_FILE
  27. static char THIS_FILE[] = __FILE__;
  28. #endif
  29.  
  30. char const g_szFilter[] = "Icon files (*.ico)|*.ico|Executables (*.exe; *.dll)|*.exe; *.dll|All Files (*.*)|*.*||";
  31. char const g_szLogFilter[] = "Log files (*.log)|*.log|Text files (*.txt)|*.txt|All Files (*.*)|*.*||";
  32. char const g_szCustIconTitle[] = "Please select an icon file";
  33. char const g_szLogTitle[] = "Choose a name for the log file";
  34. char const g_szViewLogTitle[] = "Please select a log file";
  35.  
  36. /////////////////////////////////////////////////////////////////////////////
  37. // CFavOrgView
  38.  
  39. IMPLEMENT_DYNCREATE(CFavOrgView, CFormView)
  40.  
  41. BEGIN_MESSAGE_MAP(CFavOrgView, CFormView)
  42.     //{{AFX_MSG_MAP(CFavOrgView)
  43.     ON_BN_CLICKED(IDC_ALLOWUPDATE, OnReplaceUrlAllowed)
  44.     ON_COMMAND(IDC_ASSIGNTHIS, OnAssignThis)
  45.     ON_BN_CLICKED(IDC_CANCELACTION, OnCancelAction)
  46.     ON_COMMAND(IDC_CUSTOMICON, OnCustomIcon)
  47.     ON_COMMAND(IDC_DELETEBYSTATE, OnDeleteByState)
  48.     ON_COMMAND(IDC_GENERATELOG, OnGenerateLog)
  49.     ON_BN_CLICKED(IDC_GO, OnGo)
  50.     ON_COMMAND(IDC_EDITURL, OnEditUrl)
  51.     ON_COMMAND(IDC_OPTIONS, OnOptions)
  52.     ON_COMMAND(IDC_PICKICON, OnPickIcon)
  53.     ON_BN_CLICKED(IDC_REFRESH, OnRefresh)
  54.     ON_COMMAND(IDC_REMOVE, OnRemove)
  55.     ON_BN_CLICKED(IDC_RESETTOROOT, OnResetDeadLinks)
  56.     ON_COMMAND(IDC_SHOWLEGEND, OnShowLegend)
  57.     ON_UPDATE_COMMAND_UI(IDC_ASSIGNTHIS, OnUpdateAssignThis)
  58.     ON_UPDATE_COMMAND_UI(IDC_CUSTOMICON, OnUpdateCustomIcon)
  59.     ON_UPDATE_COMMAND_UI(IDC_DELETEBYSTATE, OnUpdateDeleteByState)
  60.     ON_UPDATE_COMMAND_UI(IDC_GENERATELOG, OnUpdateGenerateLog)
  61.     ON_UPDATE_COMMAND_UI(IDC_EDITURL, OnUpdateNewUrl)
  62.     ON_UPDATE_COMMAND_UI(IDC_PICKICON, OnUpdatePickIcon)
  63.     ON_UPDATE_COMMAND_UI(IDC_REMOVE, OnUpdateRemove)
  64.     ON_UPDATE_COMMAND_UI(IDC_SHOWLEGEND, OnUpdateShowLegend)
  65.     ON_WM_PAINT()
  66.     ON_WM_DESTROY()
  67.     ON_WM_SIZE()
  68.     ON_BN_CLICKED(IDC_FINDFAVICONS, OnFindFavicons)
  69.     ON_COMMAND(IDC_DISPINFO, OnDispInfo)
  70.     ON_UPDATE_COMMAND_UI(IDC_DISPINFO, OnUpdateDispInfo)
  71.     ON_COMMAND(IDC_EXPANDALL, OnExpandAll)
  72.     ON_COMMAND(IDC_EXPANDCURRENT, OnExpandCurrent)
  73.     ON_UPDATE_COMMAND_UI(IDC_EXPANDCURRENT, OnUpdateExpandCurrent)
  74.     ON_COMMAND(IDC_COLLAPSEALL, OnCollapseAll)
  75.     ON_UPDATE_COMMAND_UI(IDC_COLLAPSEALL, OnUpdateCollapseAll)
  76.     ON_UPDATE_COMMAND_UI(IDC_EXPANDALL, OnUpdateExpandAll)
  77.     ON_COMMAND(IDC_SELECTNONE, OnSelectNone)
  78.     ON_UPDATE_COMMAND_UI(IDC_SELECTNONE, OnUpdateSelectNone)
  79.     ON_COMMAND(IDC_COLLAPSECURRENT, OnCollapseCurrent)
  80.     ON_UPDATE_COMMAND_UI(IDC_COLLAPSECURRENT, OnUpdateCollapseCurrent)
  81.     ON_MESSAGE(WM_IDLEUPDATECMDUI, OnIdleUpdateCmdUI)
  82.     ON_UPDATE_COMMAND_UI(IDC_CANCELACTION, OnUpdateCancelAction)
  83.     ON_UPDATE_COMMAND_UI(IDC_ALLOWUPDATE, OnUpdateAllowUpdate)
  84.     ON_UPDATE_COMMAND_UI(IDC_FINDFAVICONS, OnUpdateFindFavIcons)
  85.     ON_UPDATE_COMMAND_UI(IDC_GO, OnUpdateGo)
  86.     ON_UPDATE_COMMAND_UI(IDC_REFRESH, OnUpdateRefresh)
  87.     ON_UPDATE_COMMAND_UI(IDC_RESETTOROOT, OnUpdateResetDeadLinks)
  88.     ON_MESSAGE(WM_CAN_UNLOCK, OnUnlock)
  89.     ON_MESSAGE(WM_DISP_MSG, OnDispMsg)
  90.     ON_MESSAGE(WM_DISP_ICON, OnDispIcon)
  91.     ON_MESSAGE(WM_REFRESH_COUNT, OnRefreshCount)
  92.     ON_COMMAND(IDC_ASSIGNCUST, OnCustomIcon)
  93.     ON_UPDATE_COMMAND_UI(IDC_ASSIGNCUST, OnUpdateCustomIcon)
  94.     ON_WM_VSCROLL()
  95.     ON_COMMAND(IDC_VIEWREPORT, OnViewReport)
  96.     ON_UPDATE_COMMAND_UI(IDC_VIEWREPORT, OnUpdateViewReport)
  97.     //}}AFX_MSG_MAP
  98.     ON_COMMAND_RANGE(IDC_MNU_QUESTION, IDC_MNU_RESET, OnStatus)
  99.     ON_UPDATE_COMMAND_UI_RANGE(IDC_MNU_QUESTION, IDC_MNU_RESET, OnUpdateStatus)
  100. END_MESSAGE_MAP()
  101.  
  102. /////////////////////////////////////////////////////////////////////////////
  103. // CFavOrgView construction/destruction
  104.  
  105. CFavOrgView::CFavOrgView()
  106.     : CFormView(CFavOrgView::IDD)
  107. {
  108.     //{{AFX_DATA_INIT(CFavOrgView)
  109.     //}}AFX_DATA_INIT
  110.     // TODO: add construction code here
  111.  
  112.     m_pdlgLegend = NULL;
  113.     m_bInitialUpdate = FALSE;
  114.     m_bLocked = FALSE;
  115.     m_bSizeInit = FALSE;
  116. }
  117.  
  118. CFavOrgView::~CFavOrgView()
  119. {
  120. }
  121.  
  122. void CFavOrgView::DoDataExchange(CDataExchange* pDX)
  123. {
  124.     CFormView::DoDataExchange(pDX);
  125.     //{{AFX_DATA_MAP(CFavOrgView)
  126.     DDX_Control(pDX, IDC_PROGRESS, m_ctlProgress);
  127.     //}}AFX_DATA_MAP
  128. }
  129.  
  130. BOOL CFavOrgView::PreCreateWindow(CREATESTRUCT& cs)
  131. {
  132.     // TODO: Modify the Window class or styles here by modifying
  133.     //  the CREATESTRUCT cs
  134.  
  135.     return CFormView::PreCreateWindow(cs);
  136. }
  137.  
  138. void CFavOrgView::OnPaint() 
  139. {
  140.     CPaintDC dc(this); // device context for painting
  141.     
  142.     if (!m_bInitialUpdate)
  143.     {
  144.         m_bInitialUpdate = TRUE;
  145.         // A trick to start painting before initiliazation is done
  146.         // Thanks to Mike Blaszczak
  147.         FakeInitialUpdate();
  148.     }
  149.     
  150.     // Do not call CFormView::OnPaint() for painting messages
  151. }
  152.  
  153. void CFavOrgView::FakeInitialUpdate()
  154. {
  155.     ShowWindow(SW_SHOW);
  156.     UpdateWindow();
  157.  
  158.     DisableCommands(TRUE);
  159.     // Load the Favorites tree
  160.     m_ctlFavTree.LoadFavorites(((CFavOrgApp*) AfxGetApp())->m_sIniFile);
  161.     DisableCommands(FALSE);
  162.  
  163.     DisplayFavNum();
  164. }
  165.  
  166. void CFavOrgView::OnInitialUpdate()
  167. {
  168.     CFormView::OnInitialUpdate();
  169.  
  170.     m_pStatusBar = ((CMainFrame*) AfxGetMainWnd())->GetStatusBar();
  171.     m_pStatusBar->UpdateWindow();
  172.  
  173.     GetParentFrame()->RecalcLayout();
  174.     ResizeParentToFit(FALSE);
  175.  
  176.     // Don't want to see the scrollbars
  177.     SetScrollSizes(MM_TEXT, CSize(0, 0));
  178.  
  179.     // Subclass our standard Tree control
  180.     m_ctlFavTree.SubclassDlgItem(IDC_FAVTREE, this);
  181.     GetDocument()->SetTreeCtrl(&m_ctlFavTree);
  182.     GetDocument()->FinalInit();
  183.  
  184.     // Give our CFavOrgTreeCtl the address of interface elements
  185.     m_ctlFavTree.SetProgressBar(&m_ctlProgress);
  186.  
  187.     // Store initial data for resizing
  188.     CRect    rectView;
  189.     CRect    rectGo;
  190.     CRect    rectFind;
  191.     CRect    rectRefresh;
  192.     CRect    rectTree;
  193.     CRect    rectProgress;
  194.  
  195.     CWnd*    pwndGo = GetDlgItem(IDC_GO);
  196.     CWnd*    pwndFind = GetDlgItem(IDC_FINDFAVICONS);
  197.     CWnd*    pwndRefresh = GetDlgItem(IDC_REFRESH);
  198.  
  199.     GetClientRect(rectView);
  200.     ClientToScreen(&rectView);
  201.  
  202.     pwndGo->GetWindowRect(rectGo);
  203.     pwndFind->GetWindowRect(rectFind);
  204.     pwndRefresh->GetWindowRect(rectRefresh);
  205.  
  206.     m_ctlFavTree.GetWindowRect(rectTree);
  207.     m_ctlProgress.GetWindowRect(rectProgress);
  208.  
  209.     m_nRightOffset1 = rectView.right - rectGo.left;
  210.     m_nRightOffset2 = rectView.right - rectFind.left;
  211.     m_nBottomOffset = rectView.bottom - rectRefresh.top;
  212.     m_nTreeOffsetH = rectView.right - rectTree.right;
  213.     m_nTreeOffsetV = rectView.bottom - rectTree.bottom;
  214.     m_nProgressOffsetH = rectView.right - rectProgress.right;
  215.     m_nProgressOffsetVTop = rectView.bottom - rectProgress.top;
  216.     m_nProgressOffsetVBottom = rectView.bottom - rectProgress.bottom;
  217.     m_bSizeInit = TRUE;
  218. }
  219.  
  220. void CFavOrgView::OnSize(UINT nType, int cx, int cy) 
  221. {
  222.     if (m_bSizeInit)
  223.     {
  224.         // Store initial data for resizing
  225.         CRect    rectView;
  226.         CRect    rectGo;
  227.         CRect    rectCancel;
  228.         CRect    rectFind;
  229.         CRect    rectOption;
  230.         CRect    rectAllow;
  231.         CRect    rectRefresh;
  232.         CRect    rectReset;
  233.         CRect    rectTree;
  234.         CRect    rectProgress;
  235.  
  236.         CWnd*    pwndGo = GetDlgItem(IDC_GO);
  237.         CWnd*    pwndCancel = GetDlgItem(IDC_CANCELACTION);
  238.         CWnd*    pwndOption = GetDlgItem(IDC_OPTIONGROUP);
  239.         CWnd*    pwndFind = GetDlgItem(IDC_FINDFAVICONS);
  240.         CWnd*    pwndAllow = GetDlgItem(IDC_ALLOWUPDATE);
  241.         CWnd*    pwndRefresh = GetDlgItem(IDC_REFRESH);
  242.         CWnd*    pwndReset = GetDlgItem(IDC_RESETTOROOT);
  243.  
  244.         GetClientRect(rectView);
  245.  
  246.         pwndGo->GetWindowRect(rectGo);
  247.         ScreenToClient(&rectGo);
  248.         pwndCancel->GetWindowRect(rectCancel);
  249.         ScreenToClient(&rectCancel);
  250.         pwndOption->GetWindowRect(rectOption);
  251.         ScreenToClient(&rectOption);
  252.         pwndFind->GetWindowRect(rectFind);
  253.         ScreenToClient(&rectFind);
  254.         pwndAllow->GetWindowRect(rectAllow);
  255.         ScreenToClient(&rectAllow);
  256.         
  257.         pwndRefresh->GetWindowRect(rectRefresh);
  258.         ScreenToClient(&rectRefresh);
  259.         pwndReset->GetWindowRect(rectReset);
  260.         ScreenToClient(&rectReset);
  261.  
  262.         m_ctlFavTree.GetWindowRect(rectTree);
  263.         ScreenToClient(&rectTree);
  264.         m_ctlProgress.GetWindowRect(rectProgress);
  265.         ScreenToClient(&rectProgress);
  266.  
  267.         pwndGo->MoveWindow(rectView.right - m_nRightOffset1,
  268.                            rectGo.top,
  269.                            rectGo.Width(),
  270.                            rectGo.Height());
  271.         pwndCancel->MoveWindow(rectView.right - m_nRightOffset1,
  272.                            rectCancel.top,
  273.                            rectCancel.Width(),
  274.                            rectCancel.Height());
  275.         pwndOption->MoveWindow(rectView.right - m_nRightOffset1,
  276.                            rectOption.top,
  277.                            rectOption.Width(),
  278.                            rectOption.Height());
  279.         pwndFind->MoveWindow(rectView.right - m_nRightOffset2,
  280.                            rectFind.top,
  281.                            rectFind.Width(),
  282.                            rectFind.Height());
  283.         pwndAllow->MoveWindow(rectView.right - m_nRightOffset2,
  284.                            rectAllow.top,
  285.                            rectAllow.Width(),
  286.                            rectAllow.Height());
  287.  
  288.         pwndRefresh->MoveWindow(rectRefresh.left,
  289.                            rectView.bottom - m_nBottomOffset,
  290.                            rectRefresh.Width(),
  291.                            rectRefresh.Height());
  292.         pwndReset->MoveWindow(rectReset.left,
  293.                            rectView.bottom - m_nBottomOffset,
  294.                            rectReset.Width(),
  295.                            rectReset.Height());
  296.  
  297.         rectTree.right = rectView.right - m_nTreeOffsetH;
  298.         rectTree.bottom = rectView.bottom - m_nTreeOffsetV;
  299.         m_ctlFavTree.MoveWindow(&rectTree);
  300.  
  301.         rectProgress.right = rectView.right - m_nProgressOffsetH;
  302.         rectProgress.top = rectView.bottom - m_nProgressOffsetVTop;
  303.         rectProgress.bottom = rectView.bottom - m_nProgressOffsetVBottom;
  304.         m_ctlProgress.MoveWindow(&rectProgress);
  305.     }
  306.  
  307.     CFormView::OnSize(nType, cx, cy);
  308. }
  309.  
  310. LRESULT CFavOrgView::OnUnlock(WPARAM wParam, LPARAM lParam)
  311. {
  312.     DisableCommands(FALSE);
  313.     
  314.     return(0);
  315. }
  316.  
  317. LRESULT CFavOrgView::OnDispMsg(WPARAM wParam, LPARAM lParam)
  318. {
  319.     m_pStatusBar->SetPaneText(2, lParam == NULL ? "" : (LPCSTR) lParam);
  320.     return(0);
  321. }
  322.  
  323. LRESULT CFavOrgView::OnDispIcon(WPARAM wParam, LPARAM lParam)
  324. {
  325.     m_pStatusBar->GetStatusBarCtrl().SetIcon(1, (HICON) lParam);
  326.     return(0);
  327. }
  328.  
  329. LRESULT CFavOrgView::OnRefreshCount(WPARAM wParam, LPARAM lParam)
  330. {
  331.     DisplayFavNum();
  332.     return(0);
  333. }
  334.  
  335. /////////////////////////////////////////////////////////////////////////////
  336. // CFavOrgView diagnostics
  337.  
  338. #ifdef _DEBUG
  339. void CFavOrgView::AssertValid() const
  340. {
  341.     CFormView::AssertValid();
  342. }
  343.  
  344. void CFavOrgView::Dump(CDumpContext& dc) const
  345. {
  346.     CFormView::Dump(dc);
  347. }
  348.  
  349. CFavOrgDoc* CFavOrgView::GetDocument() // non-debug version is inline
  350. {
  351.     ASSERT(m_pDocument->IsKindOf(RUNTIME_CLASS(CFavOrgDoc)));
  352.     return (CFavOrgDoc*)m_pDocument;
  353. }
  354. #endif //_DEBUG
  355.  
  356. /////////////////////////////////////////////////////////////////////////////
  357. // CFavOrgView helpers
  358.  
  359. void CFavOrgView::DisplayFavNum()
  360. {
  361.     // Display number of favorites
  362.     CString sFavNum, sNumFmt;
  363.     
  364.     sNumFmt.LoadString(IDS_NUMFMT);
  365.     sFavNum.Format(sNumFmt, m_ctlFavTree.GetItemCount());
  366.     SetDlgItemText(IDC_FAVNUMBER, sFavNum);
  367. }
  368.  
  369.  
  370. /////////////////////////////////////////////////////////////////////////////
  371. // CFavOrgView message handlers
  372.  
  373.  
  374. void CFavOrgView::OnReplaceUrlAllowed() 
  375. {
  376.     BOOL bAllowed = (BOOL) ((CButton*) GetDlgItem(IDC_ALLOWUPDATE))->GetCheck();
  377.     m_ctlFavTree.SetRedirFlag(bAllowed);
  378.     GetDocument()->m_nAllowUpdate = bAllowed ? 1 : 0;
  379. }
  380.  
  381. void CFavOrgView::OnFindFavicons() 
  382. {
  383.     BOOL bFindIcons = (BOOL) ((CButton*) GetDlgItem(IDC_FINDFAVICONS))->GetCheck();
  384.     GetDocument()->m_nFindIcons = bFindIcons ? 1 : 0;
  385. }
  386.  
  387. void CFavOrgView::OnAssignThis() 
  388. {
  389.     // Assign picked icon to selected items
  390.     m_ctlFavTree.AssignThisImage(NULL);    
  391. }
  392.  
  393. void CFavOrgView::OnCancelAction() 
  394. {
  395.     // Cancel current action
  396.     m_ctlFavTree.CancelAction();
  397. }
  398.  
  399. void CFavOrgView::OnCustomIcon() 
  400. {
  401.     m_ctlFavTree.ExecLocalCommand(IDC_ASSIGNCUST, FALSE);
  402. }
  403.  
  404. void CFavOrgView::OnDispInfo() 
  405. {
  406.     m_ctlFavTree.ExecLocalCommand(IDC_DISPINFO, FALSE);
  407. }
  408.  
  409. void CFavOrgView::OnDeleteByState() 
  410. {
  411.     DisableCommands(TRUE);
  412.     m_ctlFavTree.DeleteByState();    
  413.     DisableCommands(FALSE);
  414. }
  415.  
  416. void CFavOrgView::OnGenerateLog() 
  417. {
  418.     // Prepare report generation
  419.     CFileDialog dlg(FALSE, 
  420.                     "*.log", 
  421.                     NULL, 
  422.                     OFN_HIDEREADONLY | OFN_OVERWRITEPROMPT | OFN_PATHMUSTEXIST, 
  423.                     g_szLogFilter,
  424.                     this);
  425.  
  426.     dlg.m_ofn.lpstrTitle = g_szLogTitle;
  427.  
  428.     CStringEx sInitialDir = ((CFavOrgApp*) AfxGetApp())->m_sIniFile;
  429.  
  430.     sInitialDir.StripToPath();
  431.     dlg.m_ofn.lpstrInitialDir = sInitialDir;
  432.  
  433.     if (dlg.DoModal() == IDOK)
  434.         m_ctlFavTree.GenerateLog(dlg.GetPathName());
  435. }
  436.  
  437. void CFavOrgView::OnViewReport() 
  438. {
  439.     CFileDialog dlg(TRUE, 
  440.                     "*.log", 
  441.                     NULL, 
  442.                     OFN_HIDEREADONLY | OFN_OVERWRITEPROMPT | OFN_PATHMUSTEXIST, 
  443.                     g_szLogFilter,
  444.                     this);
  445.  
  446.     dlg.m_ofn.lpstrTitle = g_szViewLogTitle;
  447.     CStringEx sInitialDir = ((CFavOrgApp*) AfxGetApp())->m_sIniFile;
  448.  
  449.     sInitialDir.StripToPath();
  450.     dlg.m_ofn.lpstrInitialDir = sInitialDir;
  451.  
  452.     if (dlg.DoModal() == IDOK)
  453.     {
  454.         CString sCommand = "notepad.exe ";
  455.  
  456.         sCommand += dlg.GetPathName();
  457.         WinExec(sCommand, SW_SHOWNORMAL);
  458.     }
  459. }
  460.  
  461. void CFavOrgView::OnGo() 
  462. {
  463.      //m_ctlFavTree.GetConnectState(TRUE);
  464.  
  465.     DisableCommands(TRUE);
  466.     // Find dead links and icons
  467.     m_ctlFavTree.CheckAndFindIcons(!(BOOL) ((CButton*) GetDlgItem(IDC_FINDFAVICONS))->GetCheck());
  468.     DisableCommands(FALSE);
  469. }
  470.  
  471. void CFavOrgView::OnEditUrl() 
  472. {
  473.     m_ctlFavTree.UpdateUrl(NULL);
  474. }
  475.  
  476. void CFavOrgView::OnOptions() 
  477. {
  478.     COptionsDlg dlg;
  479.  
  480.     dlg.m_sStorePath = GetDocument()->m_sStorePath;
  481.     dlg.m_nTimeout = GetDocument()->m_nTimeout;
  482.     dlg.m_bNoConnectiontest = GetDocument()->m_nNoConnectTest == 1;
  483.     dlg.m_bCanSetFolder = !GetDocument()->m_bCommandsDisabled;
  484.  
  485.     if (dlg.DoModal() == IDOK)
  486.     {
  487.         GetDocument()->m_sStorePath = dlg.m_sStorePath;
  488.         m_ctlFavTree.SetIconStore(dlg.m_sStorePath);
  489.         GetDocument()->m_nTimeout = dlg.m_nTimeout;
  490.         m_ctlFavTree.SetTimeout(dlg.m_nTimeout);
  491.         GetDocument()->m_nNoConnectTest = dlg.m_bNoConnectiontest ? 1 : 0;
  492.         m_ctlFavTree.SetConnectCheck(dlg.m_bNoConnectiontest);
  493.     }
  494. }
  495.  
  496. void CFavOrgView::OnPickIcon() 
  497. {
  498.     // Pick icon of selected link
  499.     m_ctlFavTree.PickCurrentIcon(NULL);
  500. }
  501.  
  502. void CFavOrgView::OnStatus(UINT nID)
  503. {
  504.     m_ctlFavTree.ExecLocalCommand(nID, FALSE);
  505. }
  506.  
  507. void CFavOrgView::OnRefresh() 
  508. {
  509.     DisableCommands(TRUE);
  510.     // Refresh tree
  511.     m_ctlFavTree.LoadFavorites(((CFavOrgApp*) AfxGetApp())->m_sIniFile);
  512.     DisableCommands(FALSE);
  513.     DisplayFavNum();
  514. }
  515.  
  516. void CFavOrgView::OnRemove() 
  517. {
  518.     // Send selected items to Recycle Bin
  519.     DisableCommands(TRUE);
  520.     m_ctlFavTree.RemoveSelected();    
  521.     DisableCommands(FALSE);
  522.     DisplayFavNum();
  523. }
  524.  
  525. void CFavOrgView::OnResetDeadLinks() 
  526. {
  527.     DisableCommands(TRUE);
  528.     // Reset dead links to root
  529.     m_ctlFavTree.ResetDeadLinks();
  530.     DisableCommands(FALSE);
  531. }
  532.  
  533. void CFavOrgView::ShowLegend(BOOL bShow)
  534. {
  535.     // Display legend dialog box as a modeless dialog
  536.     if (GetDocument()->m_bShowLegend)
  537.     {
  538.         // Display legend
  539.         if (!m_pdlgLegend)    // Dialog not created
  540.         {
  541.             m_pdlgLegend = new CLegendDlg;
  542.             m_pdlgLegend->Create(IDD_LEGEND, this);
  543.  
  544.             CRect rectLegend;
  545.  
  546.             if (GetDocument()->m_rectLegend.right != 0)
  547.                 m_pdlgLegend->MoveWindow(&(GetDocument()->m_rectLegend));
  548.             else
  549.             {
  550.                 m_pdlgLegend->GetWindowRect(&rectLegend);
  551.                 m_pdlgLegend->MoveWindow(0,
  552.                                          0,
  553.                                          rectLegend.right - rectLegend.left,
  554.                                          rectLegend.bottom - rectLegend.top);
  555.             }
  556.             m_pdlgLegend->ShowWindow(SW_SHOWNORMAL);
  557.             m_pdlgLegend->UpdateWindow();
  558.         }
  559.         else    // Just show it
  560.             m_pdlgLegend->ShowWindow(SW_SHOWNORMAL);
  561.     }
  562.     else if (m_pdlgLegend)
  563.     {
  564.         m_pdlgLegend->ShowWindow(SW_HIDE);
  565.     }
  566. }
  567.  
  568. void CFavOrgView::OnShowLegend() 
  569. {
  570.     GetDocument()->m_bShowLegend = !GetDocument()->m_bShowLegend;
  571.     ShowLegend(GetDocument()->m_bShowLegend);
  572. }
  573.  
  574. void CFavOrgView::OnExpandAll() 
  575. {
  576.     m_ctlFavTree.ExpandAll();
  577. }
  578.  
  579. void CFavOrgView::OnExpandCurrent() 
  580. {
  581.     m_ctlFavTree.ExpandCurrent();
  582. }
  583.  
  584. void CFavOrgView::OnCollapseAll() 
  585. {
  586.     m_ctlFavTree.CollapseAll();
  587. }
  588.  
  589. void CFavOrgView::OnCollapseCurrent() 
  590. {
  591.     m_ctlFavTree.CollapseCurrent();
  592. }
  593.  
  594. void CFavOrgView::OnSelectNone() 
  595. {
  596.     m_ctlFavTree.ClearSelection(TRUE);
  597. }
  598.  
  599. /////////////////////////////////////////////////////////////////////////////
  600. // CFavOrgView message handlers for UPDATE_COMMAND_UI messages
  601.  
  602. void CFavOrgView::DisableCommands(BOOL bDisable)
  603. {
  604.     m_bLocked = bDisable;
  605.     GetDocument()->m_bCommandsDisabled = bDisable;
  606.     UpdateDialogControls(this, FALSE);
  607. }
  608.  
  609. void CFavOrgView::OnUpdateAssignThis(CCmdUI* pCmdUI) 
  610. {
  611.     pCmdUI->Enable(!GetDocument()->m_bCommandsDisabled && m_ctlFavTree.IsPickedIcon());
  612. }
  613.  
  614. void CFavOrgView::OnUpdateCustomIcon(CCmdUI* pCmdUI) 
  615. {
  616.     BOOL bDummy, bDummy2;
  617.     pCmdUI->Enable(!GetDocument()->m_bCommandsDisabled && m_ctlFavTree.SelectionExist(bDummy, bDummy2));
  618. }
  619.  
  620. void CFavOrgView::OnUpdateDeleteByState(CCmdUI* pCmdUI) 
  621. {
  622.     pCmdUI->Enable(!GetDocument()->m_bCommandsDisabled);
  623. }
  624.  
  625. void CFavOrgView::OnUpdateGenerateLog(CCmdUI* pCmdUI) 
  626. {
  627.     pCmdUI->Enable(!GetDocument()->m_bCommandsDisabled);
  628. }
  629.  
  630. void CFavOrgView::OnUpdateNewUrl(CCmdUI* pCmdUI) 
  631. {
  632.     BOOL bOneShortcut, bOneFolder, bOK;
  633.  
  634.     bOK = m_ctlFavTree.SelectionExist(bOneShortcut, bOneFolder);
  635.  
  636.     pCmdUI->Enable(!GetDocument()->m_bCommandsDisabled && bOK && bOneShortcut);
  637. }
  638.  
  639. void CFavOrgView::OnUpdatePickIcon(CCmdUI* pCmdUI) 
  640. {
  641.     BOOL bOneShortcut, bOneFolder, bOK;
  642.  
  643.     bOK = m_ctlFavTree.SelectionExist(bOneShortcut, bOneFolder);
  644.  
  645.     pCmdUI->Enable(!GetDocument()->m_bCommandsDisabled && bOK && bOneShortcut);
  646. }
  647.  
  648. void CFavOrgView::OnUpdateRemove(CCmdUI* pCmdUI) 
  649. {
  650.     BOOL bDummy, bDummy2;
  651.     pCmdUI->Enable(!GetDocument()->m_bCommandsDisabled && m_ctlFavTree.SelectionExist(bDummy, bDummy2));
  652. }
  653.  
  654. void CFavOrgView::OnUpdateShowLegend(CCmdUI* pCmdUI) 
  655. {
  656.     pCmdUI->SetCheck(GetDocument()->m_bShowLegend ? 1 : 0);
  657. }
  658.  
  659. // Necessary to update controls in the form view
  660. LRESULT CFavOrgView::OnIdleUpdateCmdUI(WPARAM wParam, LPARAM)
  661. {
  662.     UpdateDialogControls(this, FALSE);
  663.     return 0L;
  664. }
  665.  
  666. void CFavOrgView::OnUpdateCancelAction(CCmdUI* pCmdUI) 
  667. {
  668.     pCmdUI->Enable(GetDocument()->m_bCommandsDisabled);
  669. }
  670.  
  671. void CFavOrgView::OnUpdateAllowUpdate(CCmdUI* pCmdUI) 
  672. {
  673.     pCmdUI->Enable(!GetDocument()->m_bCommandsDisabled);
  674. }
  675.  
  676. void CFavOrgView::OnUpdateFindFavIcons(CCmdUI* pCmdUI) 
  677. {
  678.     pCmdUI->Enable(!GetDocument()->m_bCommandsDisabled);
  679. }
  680.  
  681. void CFavOrgView::OnUpdateGo(CCmdUI* pCmdUI) 
  682. {
  683.     pCmdUI->Enable(!GetDocument()->m_bCommandsDisabled);
  684. }
  685.  
  686. void CFavOrgView::OnUpdateRefresh(CCmdUI* pCmdUI) 
  687. {
  688.     pCmdUI->Enable(!GetDocument()->m_bCommandsDisabled);
  689. }
  690.  
  691. void CFavOrgView::OnUpdateResetDeadLinks(CCmdUI* pCmdUI) 
  692. {
  693.     pCmdUI->Enable(!GetDocument()->m_bCommandsDisabled);
  694. }
  695.  
  696. void CFavOrgView::OnUpdateDispInfo(CCmdUI* pCmdUI) 
  697. {
  698.     BOOL bOneShortcut, bOneFolder, bOK;
  699.  
  700.     bOK = m_ctlFavTree.SelectionExist(bOneShortcut, bOneFolder);
  701.  
  702.     if (bOK)
  703.     {
  704.         DWORD dwData = m_ctlFavTree.GetItemData(m_ctlFavTree.GetSelectedItem());
  705.  
  706.         bOK = bOK && (dwData != 0) && (dwData != (DWORD) ITEM_IS_FOLDER);
  707.     }
  708.  
  709.     pCmdUI->Enable(!GetDocument()->m_bCommandsDisabled && bOK && bOneShortcut);
  710. }
  711.  
  712. void CFavOrgView::OnUpdateCollapseAll(CCmdUI* pCmdUI) 
  713. {
  714.     pCmdUI->Enable(!GetDocument()->m_bCommandsDisabled);
  715. }
  716.  
  717. void CFavOrgView::OnUpdateExpandAll(CCmdUI* pCmdUI) 
  718. {
  719.     pCmdUI->Enable(!GetDocument()->m_bCommandsDisabled);
  720. }
  721.  
  722. void CFavOrgView::OnUpdateViewReport(CCmdUI* pCmdUI) 
  723. {
  724.     pCmdUI->Enable(!GetDocument()->m_bCommandsDisabled);
  725. }
  726.  
  727. void CFavOrgView::OnUpdateCollapseCurrent(CCmdUI* pCmdUI) 
  728. {
  729.     BOOL bOneShortcut, bOneFolder, bOK;
  730.  
  731.     bOK = m_ctlFavTree.SelectionExist(bOneShortcut, bOneFolder);
  732.  
  733.     if (bOK)
  734.     {
  735.         DWORD dwData = m_ctlFavTree.GetItemData(m_ctlFavTree.GetSelectedItem());
  736.  
  737.         bOK = bOK && (dwData != 0) && (dwData == (DWORD) ITEM_IS_FOLDER);
  738.     }
  739.  
  740.     pCmdUI->Enable(!GetDocument()->m_bCommandsDisabled && bOK && bOneFolder);
  741. }
  742.  
  743. void CFavOrgView::OnUpdateExpandCurrent(CCmdUI* pCmdUI) 
  744. {
  745.     BOOL bOneShorcut, bOneFolder, bOK;
  746.  
  747.     bOK = m_ctlFavTree.SelectionExist(bOneShorcut, bOneFolder);
  748.  
  749.     if (bOK)
  750.     {
  751.         DWORD dwData = m_ctlFavTree.GetItemData(m_ctlFavTree.GetSelectedItem());
  752.  
  753.         bOK = bOK && (dwData != 0) && (dwData == (DWORD) ITEM_IS_FOLDER);
  754.     }
  755.  
  756.     pCmdUI->Enable(!GetDocument()->m_bCommandsDisabled && bOK && bOneFolder);
  757. }
  758.  
  759. void CFavOrgView::OnUpdateSelectNone(CCmdUI* pCmdUI) 
  760. {
  761.     BOOL bDummy, bDummy2;
  762.     pCmdUI->Enable(!GetDocument()->m_bCommandsDisabled && m_ctlFavTree.SelectionExist(bDummy, bDummy2));
  763. }
  764.  
  765. void CFavOrgView::OnUpdateStatus(CCmdUI* pCmdUI)
  766. {
  767.     int nCommonStatus = m_ctlFavTree.GetCommonStatus(NULL);
  768.  
  769.     pCmdUI->SetCheck(0);
  770.     if (nCommonStatus != -1)
  771.     {
  772.         int nMenuToCheck = m_ctlFavTree.GetMenuToCheckFromState(nCommonStatus);
  773.  
  774.         if ((nMenuToCheck != -1) && (pCmdUI->m_nID == (UINT) nMenuToCheck))
  775.             pCmdUI->SetCheck();
  776.     }
  777.  
  778.     pCmdUI->Enable(!GetDocument()->m_bCommandsDisabled);
  779. }
  780.  
  781. void CFavOrgView::OnDestroy()
  782. {
  783.     if (m_pdlgLegend)
  784.     {
  785.         m_pdlgLegend->DestroyWindow();
  786.         delete m_pdlgLegend;
  787.     }
  788.  
  789.     CFormView::OnDestroy();
  790. }
  791.  
  792. BOOL CFavOrgView::OnCommand(WPARAM wParam, LPARAM lParam) 
  793. {
  794.     // The user closed the legend window
  795.     if (wParam == CMD_LEGEND_CLOSING)
  796.     {
  797.         GetDocument()->m_bShowLegend = FALSE;
  798.         m_pdlgLegend->GetWindowRect(&(GetDocument()->m_rectLegend));
  799.         m_pdlgLegend->DestroyWindow();
  800.         delete m_pdlgLegend;
  801.         m_pdlgLegend = NULL;
  802.  
  803.         return TRUE;
  804.     }
  805.  
  806.     return CFormView::OnCommand(wParam, lParam);
  807. }
  808.  
  809.